#!/bin/bash
gpu=$1
ARGS=${@:2}

python train.py \
  --exp Pretrain \
  --model-path saved \
  --tb-path tensorboard \
  --gpu $gpu \
  $ARGS


python train.py --exp Pretrain --model-path saved --tb-path tensorboard --gpu <gpu> --batch-size 256
